package test01;

import java.lang.reflect.Field;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;

/**
 * @Author czl
 * @Date 2024-04-07
 */
public class SqlRunner {
    private Connection connection;

    public SqlRunner() {
    }

    public SqlRunner(Connection connection) {
        this.connection = connection;
    }

    /**
     * 通用的更新操作方法 - 增、删、改
     *
     * @param sql    增、删、改的 SQL 语句
     * @param params SQL语句需要的具体数据 - 不定长参数本质是一个数组
     * @return 受影响的行数
     */
    public int update(String sql,Object ... params){
        int row = 0;

        // 第一：定义要操作的SQL语句
        // 第二：获取连接对象

        PreparedStatement ps = null;
        try {
            // 第三：预编译SQL语句
            ps = connection.prepareStatement(sql);
            // 第四：填充参数 - 循环遍历不定长参数，调用 ps.setObject(下标,值)
            // insert into staffs values (?,?,?,?,?,?,?) ;
            setParameters(params, ps);
            // 第五：执行SQL语句
            row = ps.executeUpdate();
        } catch (SQLException e) {
            throw new RuntimeException(e);
        } finally {
            // 第六：关闭对象
            DbUtil.close(null, ps, connection);
        }
        return row ;
    }

    /**
     * 填充占位符 - ctrl + alt + m
     * 1）单一职责原则
     * 2）复用性 - 一次定义，多次使用
     *
     *
     * @param params
     * @param ps
     * @throws SQLException
     */
    private static void setParameters(Object[] params, PreparedStatement ps) throws SQLException {
        for (int i = 0; i < params.length; i++) {
            ps.setObject(i + 1, params[i]);
        }
    }

    /**
     * 通过查询操作 - 查询任何数据表，返回 T 类型的 List 集合
     *
     * @param entity 实体对象
     * @param sql    查询SQL语句
     * @param params SQL语句需要的具体数据
     * @param <T>    实体对象的具体类型
     * @return T 类型的 List 集合
     */
    public <T> List<T> query(Class<T> entity, String sql, Object... params) {
        // 第一：定义要操作的 SQL 语句
        // 第二：获取连接对象
        ResultSet rst = null;
        PreparedStatement ps = null;
        try {
            // 第三：预编译 SQL 语句
            ps = connection.prepareStatement(sql);

            // 第四：填充参数
            setParameters(params, ps);

            // 第五：执行查询操作 - ResultSet 对象
            rst = ps.executeQuery();

            // 第六：把 ResultSet 对象中的数据封装到 List<T> 集合返回  - 反射、内省、注解
            return beanListHandler(rst, entity);
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }finally {
            // 第七：关闭对象
            DbUtil.close(rst, ps, connection);
        }
    }

    private <T> List<T> beanListHandler(ResultSet rst, Class<T> entityClass) {
        List<T> list = new ArrayList<>();
        try {
            // 第一：获取结果集元数据对象
            ResultSetMetaData metaData = rst.getMetaData();

            // 第二：获取 select 语句后面查询的字段数量 - 循环读取各列数据
            int count = metaData.getColumnCount();
            System.out.println("查询列数：" + count);

            // 第三：循环遍历结果集 - 循环各行
            while (rst.next()) {
                // 第四：实例化实体对象 - 如何赋值？
                T entity = entityClass.getConstructor().newInstance();

                // 第五：循环查询的各列
                for (int i = 0; i < count; i++) {
                    // 第六：获取 select 语句后面查询的列名称
                    String columnName = metaData.getColumnLabel(i + 1);

                    // 第七：获取各列数据
                    Object columnData = rst.getObject(i + 1);

                    // 第八：获取实体对象对应的属性的 Field 对象
                    Field filed = entityClass.getDeclaredField(columnName);

                    // 第九：设置访问权限
                    filed.setAccessible(true);

                    // 第十：把各列数据封装到实体对象中
                    filed.set(entity,columnData);
                }
                // 第十一：把实体对象添加到 List 集合中
                list.add(entity);
            }

        } catch (Exception e) {
            throw new RuntimeException(e);
        }
        return list;
    }

}
